from selenium.webdriver.support.wait import WebDriverWait

from utils import UtilsDriver


class BasePage:
    def __init__(self):
        self.driver = UtilsDriver.get_app_driver()

    def get_element(self, location):
        return WebDriverWait(self.driver, 10).until(lambda x: x.find_element(*location))


class BaseOperate:
    def input_text(self, element, text):
        element.clear()
        element.send_keys(text)
